import os
import sys

project_dir_path = os.path.abspath(os.path.join(os.path.dirname(__file__), "../"))
sys.path.append(project_dir_path)
print(project_dir_path)

from omics_bert_str_2_embeddings import EmbeddingBuilder


def main():
    with EmbeddingBuilder(model_name="Qwen/Qwen3-Embedding-8B", 
    task="embed",
    disable_log_stats=True,
    tensor_parallel_size=2,
    gpu_memory_utilization=0.9,
    instruct="Represent the Science paragraph for retrieval:") as builder:
    
        embeddings = builder.embed(["hello", "world"])
        print(embeddings)

if __name__ == "__main__":
    main()
